import os
from .base_cfg import *

# modify dataset config
DATASET_CFG = DATASET_CFG.copy()
DATASET_CFG.update({
    'type': 'inria',
    'rootdir': '/data/SemanticSegmnetation/Inria',
})
DATALOADER_CFG = DATALOADER_CFG.copy()
DATALOADER_CFG['train'].update({
    'batch_size': 1,
})

# modify segmentor config
SEGMENTOR_CFG = SEGMENTOR_CFG.copy()
SEGMENTOR_CFG.update({
    'num_classes': 2,
})

COMMON_CFG = COMMON_CFG.copy()
COMMON_CFG['work_dir'] = 'upernet_maevitbase_inria'
COMMON_CFG['logfilepath'] = 'upernet_maevitbase_inria/upernet_maevitbase_inria.log'
COMMON_CFG['resultsavepath'] = 'upernet_maevitbase_inria/upernet_maevitbase_inria_results.pkl'